bs=8
path_to_ImageNetC=/ifs/root/ipa01/101/user_101002/Datasets/Multitask_Datasets
output_dir=/ifs/root/ipa01/101/user_101002/Project/ControlVAR-main/infer_output/multi_task_c_img_2
ckpt_path=/ifs/root/ipa01/101/user_101002/Project/ControlVAR-main/experiments/experiments/multitask_8_output_4/step_69000/checkpoint_step_latest.pth
gpus=1



CUDA_VISIBLE_DEVICES=9 python3 -m torch.distributed.launch --nproc_per_node=1 --master_port=29501 train_control_var_hpu_multitask.py \
--batch_size $bs \
--dataset_name multi_dataset_task \
--data_dir $path_to_ImageNetC \
--mixed_precision no \
--gpus $gpus \
--output_dir $output_dir \
--multi_cond True \
--val_only True \
--c_img True \
--resume $ckpt_path \
--cfg 4 4 4